package com.starhub.application.model.enums;

/*
 * 模型厂商
 */
public enum ChatProviderEnums {
    starHub("0", "星枢", "星枢"),
    OpenAI("1", "openAI", "openAI"),
    Anthropic("2", "Anthropic", "Anthropic"),
    Alibaba("3", "阿里", "阿里"),
    Baidu("4", "百度", "百度"),
    XunFei("5", "讯飞", "讯飞"),
    ByteDance("6", "字节跳动", "字节跳动"),
    Deepseek("7", "deepseek", "deepseek"),

    Custom("999", "自定义", "自定义");

    private String providerKey;
    private String providerMark;
    private String providerName;


    ChatProviderEnums(String providerKey, String providerMark, String providerName) {
        this.providerKey = providerKey;
        this.providerMark = providerMark;
        this.providerName = providerName;
    }

    public String getProviderKey() {
        return providerKey;
    }

    public String getProviderMark() {
        return providerMark;
    }

    public String getProviderName() {
        return providerName;
    }

    public static ChatProviderEnums getModelSupplierByKey(String key){
        for(ChatProviderEnums item : ChatProviderEnums.values()){
            if(item.getProviderKey().equals(key)) {
                return item;
            }
        }
        return ChatProviderEnums.Custom;
    }
}
